package com.c2b.algorithm.leetcode.base;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * <a href='https://leetcode.cn/problems/minimum-index-sum-of-two-lists/'>两个列表的最小索引总和(Minimum Index Sum of Two Lists)</a>
 * <p>假设 Andy 和 Doris 想在晚餐时选择一家餐厅，并且他们都有一个表示最喜爱餐厅的列表，每个餐厅的名字用字符串表示。</p>
 * <p>你需要帮助他们用最少的索引和找出他们共同喜爱的餐厅。 如果答案不止一个，则输出所有答案并且不考虑顺序。 你可以假设答案总是存在。</p>
 *
 * <p>
 * <b>示例：</b>
 * <pre>
 * 示例 1:
 *      输入: list1 = ["Shogun", "Tapioca Express", "Burger King", "KFC"]，list2 = ["Piatti", "The Grill at Torrey Pines", "Hungry Hunter Steakhouse", "Shogun"]
 *      输出: ["Shogun"]
 *      解释: 他们唯一共同喜爱的餐厅是“Shogun”。
 *
 * 示例 2:
 *      输入:list1 = ["Shogun", "Tapioca Express", "Burger King", "KFC"]，list2 = ["KFC", "Shogun", "Burger King"]
 *      输出: ["Shogun"]
 *      解释: 他们共同喜爱且具有最小索引和的餐厅是“Shogun”，它有最小的索引和1(0+1)。
 * </pre>
 * </p>
 * <p>
 * <b>提示：</b>
 *     <ul>
 *         <li>1 <= list1.length, list2.length <= 1000</li>
 *         <li>1 <= list1[i].length, list2[i].length <= 30 </li>
 *         <li>list1[i] 和 list2[i] 由空格 ' ' 和英文字母组成。</li>
 *         <li>list1 的所有字符串都是 唯一 的。</li>
 *         <li>list2 中的所有字符串都是 唯一 的。</li>
 *     </ul>
 * </p>
 *
 * @author c2b
 * @since 2023/12/18 11:30
 */
public class LC0599MinimumIndexSumOfTwoLists_S {
    static class Solution {
        public String[] findRestaurant(String[] list1, String[] list2) {
            Map<String, Integer> cntMap = new HashMap<>();
            for (int i = 0; i < list1.length; i++) {
                cntMap.put(list1[i], i);
            }
            int ans = Integer.MAX_VALUE;
            List<String> ansList = new ArrayList<>();
            for (int j = 0; j < list2.length; j++) {
                if (cntMap.containsKey(list2[j])) {
                    Integer i = cntMap.get(list2[j]);
                    if (i + j <= ans) {
                        if (i + j < ans){
                            ansList = new ArrayList<>();
                            ans = i + j;
                        }
                        ansList.add(list2[j]);
                    }
                }
            }
            return ansList.toArray(new String[0]);
        }
    }

    public static void main(String[] args) {
        Solution solution = new Solution();
        for (String s : solution.findRestaurant(new String[]{"Shogun", "Tapioca Express", "Burger King", "KFC"}, new String[]{"Piatti", "The Grill at Torrey Pines", "Hungry Hunter Steakhouse", "Shogun"})) {
            System.out.println(s);
        }

        System.out.println("====================");

        for (String s : solution.findRestaurant(new String[]{"Shogun", "Tapioca Express", "Burger King", "KFC"}, new String[]{"KFC", "Shogun", "Burger King"})) {
            System.out.println(s);
        }

    }
}
